package br.com.webcrm.view;

import java.util.List;

import br.com.webcrm.model.Atividade;

@SuppressWarnings("serial")
public class AtividadeDataModel extends PrimeDataModel<Atividade> {

    public AtividadeDataModel() {
    }

    public AtividadeDataModel(Object data) {
        super(data);
    }
    
    @SuppressWarnings("unchecked")
	@Override
    public Atividade getRowData(String rowKey) {
        //In a real app, a more efficient way like a query by rowKey should be implemented to deal with huge data
        
        List<Atividade> atividades = (List<Atividade>) getWrappedData();
        
        for(Atividade atividade : atividades) {
        	String id = ""+atividade.getIdAtividade();
            if(id.equals(rowKey))
                return atividade;
        }
        
        return null;
    }

    @Override
    public String getRowKey(Atividade atividade) {
        return ""+atividade.getIdAtividade();
    }
}
                    